package com.forum.framework.common.pojo;

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.forum.framework.common.exception.ErrorCode;
import com.forum.framework.common.exception.ServiceException;
import com.forum.framework.common.exception.enums.GlobalErrorCodeConstants;
import org.springframework.util.Assert;

import java.io.Serializable;
import java.util.Objects;

/**
 * 通用返回结果，和 {@link DataResult} 语义相同，
 *
 * @author zihan.ouyang
 */
public class DataResult<T> implements Serializable {
    /**
     * 错误码
     *
     * @see ErrorCode#getCode()
     */
    private Integer code;
    /**
     * 返回数据
     */
    private T data;

    /**
     * 错误信息
     *
     * @see ErrorCode#getMsg()
     */
    private String msg;

    public static <T> DataResult<T> success(T data) {
        DataResult<T> result = new DataResult<>();
        result.code = GlobalErrorCodeConstants.SUCCESS.getCode();
        result.data = data;
        result.msg = "";
        return result;
    }

    private DataResult() {
    }

    private DataResult(int code, String message) {
        this.code = code;
        this.msg = message;
    }

    public static DataResult<Void> success(int code, String msg) {
        return new DataResult<>(code, msg);
    }

    /**
     * 将传入的 result 对象，转换成另外一个泛型结果的对象
     * <p>
     * 因为 A 方法返回的 CommonResult 对象，不满足调用其的 B 方法的返回，所以需要进行转换。
     *
     * @param result 传入的result 对象
     * @param <T>    返回的泛型对象
     * @return 新的 DataResult 对象
     */
    public static <T> DataResult<T> error(DataResult<?> result) {
        return error(result.code, result.msg);
    }

    public static <T> DataResult<T> error(Integer code, String msg) {
        Assert.isTrue(!GlobalErrorCodeConstants.SUCCESS.getCode().equals(code), "code 必须是错误的");
        DataResult<T> result = new DataResult<>();
        result.code = code;
        result.msg = msg;
        return result;
    }

    public static <T> DataResult<T> error(ErrorCode errorCode) {
        return error(errorCode.getCode(), errorCode.getMsg());
    }

    public boolean isSuccess(Integer code) {
        return Objects.equals(code, GlobalErrorCodeConstants.SUCCESS.getCode());
    }

    @JsonIgnore // 避免 jackson 序列化
    public boolean isSuccess() {
        return isSuccess(code);
    }

    @JsonIgnore
    public boolean isError() {
        return !isSuccess();
    }

    // ========= 和 Exception 异常体系集成 =========

    /**
     * 判断是否有异常。如果有，则抛出 {@link ServiceException} 异常
     */
    public void checkError() throws ServiceException {
        if (isSuccess()) {
            return;
        }
        throw new ServiceException(code, msg);
    }

    /**
     * 判断是否有异常，如果有，则抛出 {@link ServiceException} 异常
     * 如果没有，则返回 {@link #data}
     */
    @JsonIgnore
    public T getCheckedData() {
        checkError();
        return data;
    }

    public static <T> DataResult<T> error(ServiceException exception) {
        return error(exception.getCode(), exception.getMessage());
    }

    public Integer getCode() {
        return code;
    }

    public void setCode(Integer code) {
        this.code = code;
    }

    public T getData() {
        return data;
    }

    public void setData(T data) {
        this.data = data;
    }

    public String getMsg() {
        return msg;
    }

    public void setMsg(String msg) {
        this.msg = msg;
    }
}
